package com.kai.Controller;

import com.alibaba.excel.EasyExcel;
import com.kai.Listener.DemoDataListener;
import com.kai.Mapper.DemoDataMapper;
import com.kai.Pojo.DemoData;
import com.kai.dao.DemoDAO;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.stream.Collectors;

@RestController
public class DemodataController {

    @Resource
    private DemoDataMapper mapper;

    @Resource
    private DemoDAO demoDAO;

    @RequestMapping("/getDemoAll")
    public List<DemoData> selectGetAll(){
        List<DemoData> allData = mapper.getAllData();
        return allData;
    }


    @RequestMapping("/readDemo")
    public List<DemoData> simpleRead() {

        // 有个很重要的点 DemoDataListener 不能被spring管理，要每次读取excel都要new,然后里面用到spring可以构造方法传进去
        String fileName = "E:\\几个项目源代码\\easyExcel.xls";
        // 这里 需要指定读用哪个class去读，然后读取第一个sheet 文件流会自动关闭
        new Thread(new Runnable(){
            @Override
            public void run() {
                EasyExcel.read(fileName, DemoData.class, new DemoDataListener(demoDAO)).sheet().doRead();
            }
        }).start();

        return mapper.getAllData();
    }

    /**
     * 最简单的写
     * <p>1. 创建excel对应的实体对象 参照{@link DemoData}
     * <p>2. 直接写即可
     */
    @RequestMapping("/writeDemo")
    public void simpleWrite() {

        String fileName = "E:\\几个项目源代码\\easyExcel.xls";
        // 这里 需要指定写用哪个class去写，然后写到第一个sheet，名字为模板 然后文件流会自动关闭
        // 如果这里想使用03 则 传入excelType参数即可
        EasyExcel.write(fileName, DemoData.class).sheet("模板").doWrite(data());
    }

    //写入的数据
    private List<DemoData> data() {
        List<DemoData> list = new ArrayList<DemoData>();
        for (int i = 0; i < 20; i++) {
            DemoData data = new DemoData();
            data.setString("字符串" + i);
            data.setDate(new Date());
            data.setDoubleData(0.56);
            list.add(data);
        }
        return list;
    }
}
